//给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。 
//
// 最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。 
//
// 你可以假设除了整数 0 之外，这个整数不会以零开头。 
//
// 
//
// 示例 1： 
//
// 
//输入：digits = [1,2,3]
//输出：[1,2,4]
//解释：输入数组表示数字 123。
// 
//
// 示例 2： 
//
// 
//输入：digits = [4,3,2,1]
//输出：[4,3,2,2]
//解释：输入数组表示数字 4321。
// 
//
// 示例 3： 
//
// 
//输入：digits = [0]
//输出：[1]
// 
//
// 
//
// 提示： 
//
// 
// 1 <= digits.length <= 100 
// 0 <= digits[i] <= 9 
// 
//
// Related Topics 数组 数学 👍 1047 👎 0

package leetcode.editor.cn;

import java.util.Arrays;

public class _66_PlusOne {
    public static void main(String[] args) {
        Solution solution = new _66_PlusOne().new Solution();
//        int[] digits = new int[] {4,3,2,1};
        int[] digits = new int[] {0};
        int[] res = solution.plusOne(digits);
        System.out.println(Arrays.toString(res));
    }
    class Solution {
        public int[] plusOne(int[] digits) {
            int index = digits.length - 1;
            for (int i = digits.length - 1; i >= 0; i--) {
                if (digits[i] != 9) {
                    index = i;
                    break;
                }
            }
            if (index == 0 && digits[0] == 9) {
                int[] res = new int[digits.length + 1];
                res[0] = 1;
                return  res;
            }
            digits[index]++;
            for (int i = index + 1; i < digits.length; i++) {
                digits[i] = 0;
            }
            return digits;
        }
    }
    //leetcode submit region begin(Prohibit modification and deletion)
class Solution1 {
    public int[] plusOne(int[] digits) {
        if (digits[digits.length - 1] + 1 == 10) {
            int add = 1;
            for(int i = digits.length - 1; i >= 0; i--) {
                int tmp = digits[i];
                digits[i] = (tmp + add) % 10;
                add = (tmp + add) / 10;
            }
            if (add == 1) {
                int[] res = new int[digits.length + 1];
                res[0] = 1;
                for (int i = 0; i < digits.length; i++) {
                    res[i + 1] = digits[i];
                }
                return res;
            }
            return digits;
        } else {
            digits[digits.length - 1]++;
            return digits;
        }
    }
}
//leetcode submit region end(Prohibit modification and deletion)

}